﻿Imports System
Imports System.Text
Imports System.Diagnostics
Imports System.Runtime.InteropServices

' 取的CPU ID
Public Class clsSysInfo
    Public Declare Sub GetSystemInfo Lib "kernel32" (ByRef lpSystemInfo As SYSTEM_INFO)

    Public Structure SYSTEM_INFO
        Dim dwOemID As Integer
        Dim wProcessorArchitecture As Integer
        Dim wReserved As Integer
        Dim dwPageSize As Integer
        Dim lpMinimumApplicationAddress As Integer
        Dim lpMaximumApplicationAddress As Integer
        Dim dwActiveProcessorMask As Integer
        Dim dwNumberOrfProcessors As Integer
        Dim dwProcessorType As Integer
        Dim dwAllocationGranularity As Short
        Dim wProcessLevel As Short
        Dim wProcessorRevision As Short
    End Structure

    '取得CPU序號
    Public Shared Function GetCPUCode() As String
        Dim CPUInfo As SYSTEM_INFO
        GetSystemInfo(CPUInfo)
        Dim rtnVal As String = CPUInfo.dwProcessorType
        Console.WriteLine("*** 取得CPU序號 = {0} ***", rtnVal)
        Return rtnVal
    End Function

    '取得硬碟序號
    Public Shared Function GetHardDriveCode(ByVal DrvIdx As Byte) As String
        Dim WMI As Object = GetObject("winmgmts:")
        Dim strCls As String = "Win32_PhysicalMedia"
        Dim strKey As String = strCls & ".Tag=""\\\\.\\PHYSICALDRIVE" & DrvIdx & """"
        Dim rtnVal As String = WMI.InstancesOf(strCls)(strKey).SerialNumber.ToString.Trim
        System.Runtime.InteropServices.Marshal.ReleaseComObject(WMI)
        Return rtnVal
    End Function

    '取得主機板序號
    Public Shared Function GetMotherBoardCode() As String
        Dim WMI As Object = GetObject("winmgmts:")
        Dim strCls As String = "Win32_BaseBoard"
        Dim strKey As String = strCls & ".Tag=""Base Board"""
        Dim rtnVal As String = WMI.InstancesOf(strCls)(strKey).SerialNumber.ToString.Trim
        System.Runtime.InteropServices.Marshal.ReleaseComObject(WMI)
        Return rtnVal
    End Function
End Class

' 關閉系統
Public Class ExitWindwos
    Public Enum RestartOptions
        LogOff = 0
        PowerOff = 8
        Reboot = 2
        ShutDown = 1
        Suspend = -1
        Hibernate = -2
        'EWX_FORCE = 4 
    End Enum

    Public Structure LUID
        Dim LowPart As Integer
        Dim HighPart As Integer
    End Structure

    Public Structure LUID_AND_ATTRIBUTES

        Dim pLuid As LUID

        Dim Attributes As Integer
    End Structure

    Public Structure TOKEN_PRIVILEGES

        Dim PrivilegeCount As Integer

        Dim Privileges As LUID_AND_ATTRIBUTES
    End Structure

    Private Const TOKEN_ADJUST_PRIVILEGES = &H20
    Private Const TOKEN_QUERY = &H8
    Private Const SE_PRIVILEGE_ENABLED = &H2
    Private Const FORMAT_MESSAGE_FROM_SYSTEM = &H1000
    Private Const EWX_FORCE = 4
    Private Declare Function LoadLibrary Lib "kernel32" Alias "LoadLibraryA" (ByVal lpLibFileName As String) As IntPtr
    Private Declare Function FreeLibrary Lib "kernel32" (ByVal hLibModule As IntPtr) As Integer
    Private Declare Function GetProcAddress Lib "kernel32" (ByVal hModule As IntPtr, ByVal lpProcName As String) As IntPtr
    Private Declare Function SetSuspendState Lib "Powrprof" (ByVal Hibernate As Integer, ByVal ForceCritical As Integer, ByVal DisableWakeEvent As Integer) As Integer
    Private Declare Function OpenProcessToken Lib "advapi32.dll" (ByVal ProcessHandle As IntPtr, ByVal DesiredAccess As Integer, ByRef TokenHandle As IntPtr) As Integer
    Private Declare Function LookupPrivilegeValue Lib "advapi32.dll" Alias "LookupPrivilegeValueA" (ByVal lpSystemName As String, ByVal lpName As String, ByRef lpLuid As LUID) As Integer
    Private Declare Function AdjustTokenPrivileges Lib "advapi32.dll" (ByVal TokenHandle As IntPtr, ByVal DisableAllPrivileges As Integer, ByRef NewState As TOKEN_PRIVILEGES, ByVal BufferLength As Integer, ByRef PreviousState As TOKEN_PRIVILEGES, ByRef ReturnLength As Integer) As Integer
    Private Declare Function ExitWindowsEx Lib "user32" (ByVal uFlags As Integer, ByVal dwReserved As Integer) As Integer
    Private Declare Function FormatMessage Lib "kernel32" Alias "FormatMessageA" (ByVal dwFlags As Integer, ByVal lpSource As IntPtr, ByVal dwMessageId As Integer, ByVal dwLanguageId As Integer, ByVal lpBuffer As StringBuilder, ByVal nSize As Integer, ByVal Arguments As Integer) As Integer

    Private Shared Sub ExitWindows(ByVal how As RestartOptions, ByVal force As Boolean)
        Select Case how
            Case RestartOptions.Suspend
                SuspendSystem(False, force)
            Case RestartOptions.Hibernate
                SuspendSystem(True, force)
            Case Else
                ExitWindows(Convert.ToInt32(how), force)
        End Select
    End Sub

    Private Shared Sub ExitWindows(ByVal how As Integer, ByVal force As Boolean)
        EnableToken("SeShutdownPrivilege")
        If force Then how = how Or EWX_FORCE
        If (ExitWindowsEx(how, 0) = 0) Then Throw New PrivilegeException(FormatError(Marshal.GetLastWin32Error()))
    End Sub

    Private Shared Sub EnableToken(ByVal privilege As String)
        If Not CheckEntryPoint("advapi32.dll", "AdjustTokenPrivileges") Then Return
        Dim tokenHandle As IntPtr = IntPtr.Zero
        Dim privilegeLUID = New LUID
        Dim newPrivileges = New TOKEN_PRIVILEGES
        Dim tokenPrivileges As TOKEN_PRIVILEGES
        If (OpenProcessToken(Process.GetCurrentProcess().Handle, TOKEN_ADJUST_PRIVILEGES Or TOKEN_QUERY, tokenHandle)) = 0 Then Throw New PrivilegeException(FormatError(Marshal.GetLastWin32Error()))
        If (LookupPrivilegeValue("", privilege, privilegeLUID)) = 0 Then Throw New PrivilegeException(FormatError(Marshal.GetLastWin32Error()))
        tokenPrivileges.PrivilegeCount = 1
        tokenPrivileges.Privileges.Attributes = SE_PRIVILEGE_ENABLED
        tokenPrivileges.Privileges.pLuid = privilegeLUID
        Dim Size As Integer = 4
        If (AdjustTokenPrivileges(tokenHandle, 0, tokenPrivileges, 4 + (12 * tokenPrivileges.PrivilegeCount), newPrivileges, Size)) = 0 Then Throw New PrivilegeException(FormatError(Marshal.GetLastWin32Error()))
    End Sub

    Private Shared Sub SuspendSystem(ByVal hibernate As Boolean, ByVal force As Boolean)
        If Not CheckEntryPoint("powrprof.dll", "SetSuspendState") Then Throw New PlatformNotSupportedException("The SetSuspendState method is not supported on this system!")
        SetSuspendState(Convert.ToInt32(IIf(hibernate, 1, 0)), Convert.ToInt32(IIf(force, 1, 0)), 0)
    End Sub

    Private Shared Function CheckEntryPoint(ByVal library As String, ByVal method As String) As Boolean
        Dim libPtr As IntPtr = LoadLibrary(library)
        If Not libPtr.Equals(IntPtr.Zero) Then
            If Not GetProcAddress(libPtr, method).Equals(IntPtr.Zero) Then
                FreeLibrary(libPtr)
                Return True
            End If
            FreeLibrary(libPtr)
        End If
        Return False
    End Function

    Private Shared Function FormatError(ByVal number As Integer) As String
        Dim Buffer = New StringBuilder(255)
        FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, IntPtr.Zero, number, 0, Buffer, Buffer.Capacity, 0)
        Return Buffer.ToString()
    End Function

    '登出 
    Public Shared Sub LogOff()
        ExitWindows(RestartOptions.Reboot, False)
    End Sub

    '關閉電源 
    Public Shared Sub PowerOff()
        ExitWindows(RestartOptions.PowerOff, False)
    End Sub

    '重啟電腦 
    Public Shared Sub Reboot()
        ExitWindows(RestartOptions.Reboot, False)
    End Sub

    '關閉系統 
    Public Shared Sub ShutDown()
        ExitWindows(RestartOptions.ShutDown, False)
    End Sub

    '待機 
    Public Shared Sub Suspend()
        ExitWindows(RestartOptions.Suspend, False)
    End Sub

    '休眠 
    Public Shared Sub Hibernate()
        ExitWindows(RestartOptions.Hibernate, False)
    End Sub
End Class
Public Class PrivilegeException
    Inherits Exception

    Public Sub New()
        MyBase.New()
    End Sub

    Public Sub New(ByVal message As String)
        MyBase.New(message)
    End Sub
End Class
